from torch.utils.data import DataLoader
from data_handle.dataset import BertTextDataset

def create_data_loader(file_path, batch_size, shuffle=True):
    dataset = BertTextDataset(file_path)
    data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=shuffle)
    return data_loader
